take input as list in Python